#!/bin/bash

# 获取本机上网IP地址
IP_ADDR=$(hostname -I | awk '{print $1}')

# 运行 Docker 容器
#sudo docker run -d --device /dev/kfd --device /dev/dri -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama:rocm
sudo docker run -d --restart always --device /dev/kfd --device /dev/dri -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama:rocm
sudo docker run -d -p 3000:8080 -e OLLAMA_BASE_URL=http://'$IP_ADDR':11434 -v open-webui:/app/backend/data --name open-webui --restart always ghcr.io/open-webui/open-webui:main


# 创建启动和停止脚本
echo "创建启动和停止脚本..."
sudo bash -c 'cat << EOF > /usr/local/bin/start_ollama
#!/bin/bash
sudo docker start ollama
sudo docker start open-webui
EOF'
sudo chmod +x /usr/local/bin/start_ollama

sudo bash -c 'cat << EOF > /usr/local/bin/stop_ollama
#!/bin/bash
sudo docker stop ollama
sudo docker stop open-webui
EOF'
sudo chmod +x /usr/local/bin/stop_ollama

# 提示启动和停止 Ollama 的方法
echo "Ollama 安装完成！"
echo "启动 Ollama 运行: start_ollama"
echo "停止 Ollama 运行: stop_ollama"
